import os
import shutil

# 设置原始文件夹路径和目标保存路径
folder_path = '/media/champrin/Data/Desktop/rune dataset/extract/all2_yolo/set exhigh_exhigh'  # 替换成你的文件夹路径（原始文件夹）
save_folder = '/media/champrin/Data/Desktop/rune dataset/extract/all2_yolo/set exhigh_exhigh_exhigh'
os.makedirs(save_folder, exist_ok=True)  # 替换成你想要保存重命名文件的目标文件夹

# 设置前缀
prefix = 'exhigh_'  # 替换成你想要的前缀

# 检查目标文件夹是否存在，如果不存在则创建
if not os.path.exists(save_folder):
    os.makedirs(save_folder)

# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    if filename.endswith('.txt'):
        # 获取源文件的完整路径
        old_file_path = os.path.join(folder_path, filename)

        # 新文件名
        new_filename = prefix + filename

        # 获取新的文件完整路径（保留原文件夹，不修改源文件）
        new_file_path = os.path.join(save_folder, new_filename)

        # 复制文件到目标文件夹，并重命名
        shutil.copy(old_file_path, new_file_path)
        print(f'Copied and renamed: {filename} -> {new_filename}')


# # 遍历文件夹中的所有文件
# for filename in os.listdir(folder_path):
#     if filename.endswith('.txt'):
#         # 获取源文件的完整路径
#         old_file_path = os.path.join(folder_path, filename)
#
#         # 去除前缀后的新文件名
#         if filename.startswith(prefix):
#             new_filename = filename[len(prefix):]
#         else:
#             new_filename = filename
#
#         # 获取新的文件完整路径（保留原文件夹，不修改源文件）
#         new_file_path = os.path.join(save_folder, new_filename)
#
#         # 复制文件到目标文件夹，并重命名
#         shutil.copy(old_file_path, new_file_path)
#         print(f'Copied: {filename} -> {new_filename}')